package org.basis.algorithm.practice;

import java.util.Arrays;
import java.util.HashMap;

/**
 * 两数之和
 *
 * @author Mr_wenpan@163.com 2022/03/21 11:54
 */
public class TwoNumSum {

    public static void main(String[] args) {

    }

    /**
     * 双重循环法
     */
    public static int[] f1(int[] nums, int target) {
        int len = nums.length;
        // 直接暴力循环
        for (int i = 0; i < len; i++) {
            for (int j = i + 1; j < len; j++) {
                if (nums[j] + nums[i] == target) {
                    return new int[]{i, j};
                }
            }
        }
        return new int[0];
    }

    /**
     * 排序 + 双指针解法，时间复杂度取决于排序的时间复杂度，双指针的时间复杂度只有O（N）
     */
    public static int[] f2(int[] nums, int target) {
        // 先将数组排升序
        Arrays.sort(nums);
        int p1 = 0;
        int p2 = nums.length - 1;

        // 使用双指针遍历数组
        while (p1 < p2) {
            int temp = nums[p1] + nums[p2];
            if (temp == target) {
                return new int[]{p1, p2};
            } else if (temp > target) {
                p2--;
            } else {
                p1++;
            }
        }

        return new int[0];
    }

    /**
     * hash表解法，时间复杂度O（N），空间复杂度O（N）
     */
    public static int[] f3(int[] nums, int target) {
        // key为数值，value为该值对应的索引
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            // 检查hash表中是否存在target - nums[i]的值，如果存在则说明有符合条件的两个数
            Integer position = map.get(target - nums[i]);
            if (position != null) {
                return new int[]{position, i};
            } else {
                map.put(nums[i], i);
            }
        }
        return new int[0];
    }
}
